import os

from openai import AsyncOpenAI
from pydantic import BaseModel


class Prompt(BaseModel):
    system_prompt: str
    user_prompt: str


client = AsyncOpenAI(
    base_url=os.environ["OPENAI_API_BASE"],
    api_key=os.environ["OPENAI_API_KEY"],
)

DEFAULT_MODEL_NAME = os.getenv("MODEL_NAME")

from . import (
    clarify,
    common_extraction,
    decline,
    extraction,
    format_answer,
    llm_answer,
    rag,
    repeat,
    simplify_answer,
    summary,
    triage,
)

